"""
@Author : 合肥一元数智教育科技有限公司
@Date :  2025/3/7 15:43
@Description : 
编写一个Python程序，实现以下功能
- 创建一个名为students.csv的CSV文件，其中包含以下列标题："Name", "Age", "Score"。
- 提示用户输入若干名学生的姓名、年龄和成绩，将这些信息逐行写入students.csv文件中。
- 从students.csv文件中读取所有学生信息，并以表格形式打印到控制台
"""
import csv

students = []
while True:
    print('----------------开始录入学生信息------------------')
    student_name = input('请输入学生姓名:')
    student_age = input('请输入学生年龄:')
    student_score = input('请输入学生分数:')
    student_info = [student_name, student_age, student_score]
    students.append(student_info)
    user_choice = input('是否继续输入(Y/N)')
    if user_choice.upper() == 'Y':
        continue
    else:
        break

with open('file/student.csv', 'a', newline='') as file:
    csv_writer = csv.writer(file)
    csv_writer.writerows(students)

from prettytable import PrettyTable

# 格式化展示数据
table = PrettyTable()
with open('file/student.csv', 'r', newline='') as file:
    csv_reader = csv.reader(file)
    for index, row in enumerate(csv_reader):
        if index == 0:
            table.field_names = row
        else:
            table.add_row(row)
print(table)



